How do you rotate a square in Python